from wordpress_project.base.base_page import Tools
import time


class reg_tool(Tools):
    # 页面元素定位
    reg_url = "http://192.168.0.125:9009/wp-login.php?action=register"
    username_ele = '[name="user_login"]' # 用户名css定位
    email_ele = '[name="user_email"]' # 密码css定位
    reg_button_ele = '[name="wp-submit"]' # 登录按钮的css定位
    reg_success_ele = '[class="message"]'  # 登录成功之后的断言定位
    reg_fail_ele = '[id="login_error"]'  # 登录失败之后的断言定位

    def reg_action(self,usr,email):
        # 在方法中，调用同一个类的其他方法， self.方法名 ，在方法中，调用类的变量 self.变量名
        self.visit(self.reg_url)
        self.by_css(self.username_ele).send_keys(usr)
        time.sleep(0.5)
        self.by_css(self.email_ele).send_keys(email)
        time.sleep(0.5)
        self.by_css(self.reg_button_ele).click()
        time.sleep(0.5)

    def get_success_msg(self):
        return self.by_css(self.reg_success_ele).text

    def get_fail_msg(self):
        return self.by_css(self.reg_fail_ele).text
